import os
import sys
from src.package import FkPackage

def main():
    args = sys.argv
    # 增加一个文件路径列表，用于放置需要重构的文件
    file_list = []
    ###############################
    ####################重要！此处填写拓展包名，如果批量处理的文件夹在新月杀游戏内，会自动识别拓展包名
    #####################如果处理单独的文件，则需要手动填写拓展包名
    extension_name = "jbs"



    if len(args) < 2:
        print(f"Usage: {args[0]} <full path of lua file>")
        print(f"Or: {args[0]} --batch")
        return

    # 如果参数是--batch，则处理列表中的所有文件
    if args[1] == "--batch":

        ######################重要！
        # 在这里添加需要重构的文件路径
        prefix = r"C:\Users\tang\Desktop\study\game\test\FreeKill-release\packages\jbs"
        #要加载的文件名
        package_name = [


             "ao",
             "kuang",
             "aola",
             "luan",
             "jbsbuqun",
             "glory",
             "jbstougao",
             "jbstougao2",
             "jbstougao3",
             "jbs_backtop",
             "jbszf",
             "kuanglang",
             "dual",
             "jbsgensoul",


       

    
             
    

                       
                    ]
        for name in package_name:
            file_list.append(prefix + "/" + name + ".lua")
        if not file_list:
            print("文件列表为空，请在代码中添加文件路径")
            return
      
        opkg = FkPackage(file_list[0], True)
        opkg.renameextension(extension_name)
        opkg.create_package_file()
          # 循环处理每个文件
        for path in file_list:
            print(f"正在处理文件: {path}")
            try:
                pkg = FkPackage(path, True)
                pkg.renameextension(extension_name)
                pkg.mkSkillDir()
                newpkg = FkPackage(path, False)
                newpkg.renameextension(extension_name)
                newpkg.mkSkillDir()
                print(f"文件 {path} 处理完成")
            except Exception as e:
                print(f"处理文件 {path} 时出错: {str(e)}")
    else:
        # 单文件处理模式
        path = args[1]
        pkg = FkPackage(path, True)
        pkg.renameextension(extension_name)
        pkg.mkSkillDir()
        newpkg = FkPackage(path, False)
        newpkg.renameextension(extension_name)
        newpkg.mkSkillDir()

if __name__ == "__main__":
    main()
